import pandas as pd

# 读取原始数据
data = pd.read_csv('../maanshan_2024.csv')  # 请将 'your_data.csv' 替换为你的数据文件路径

# 转换天气类型
def simplify_weather(weather_str):
    if '晴' in weather_str:
        return '晴朗'
    elif '阴' in weather_str:
        return '阴天'
    elif '雨' in weather_str:
        return '雨天'
    elif '雪' in weather_str:
        return '雪天'
    elif '霾' in weather_str:
        return '雾霾'
    else:
        return '其他'

data['weather_type'] = data['天气'].apply(simplify_weather)

# 将“时间”列转换为 DATE 格式
data['DATE'] = pd.to_datetime(data['时间'])

# 选择需要的列
result = data[['DATE', '最高气温', '最低气温', 'weather_type']]

# 重命名列以匹配目标格式
result.columns = ['DATE', 'maxTemp', 'minTemp', 'weather_type']

# 将结果保存为 CSV 文件
result.to_csv('formatted_data1.csv', index=False)

print("数据已转换为指定格式并保存为 formatted_data.csv")